/*
 *  Copyright (c) 2016 WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 *  WSO2 Inc. licenses this file to you under the Apache License,
 *  Version 2.0 (the "License"); you may not use this file except
 *  in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing,
 *  software distributed under the License is distributed on an
 *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 *  KIND, either express or implied.  See the License for the
 *  specific language governing permissions and limitations
 *  under the License.
 *
 */
package org.wso2.msf4j.example;

import org.wso2.msf4j.Request;
import org.wso2.msf4j.Session;

import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.core.Context;

/**
 * Session-aware counter service which counts &amp; returns the number of requests from each session.
 */
@Path("/")
public class CounterService {

    private static final String COUNTER = "counter";

    @GET
    public int count(@Context Request request) {
        Session session = request.getSession();

        // Create & set the counter in the session
        Object attribute = session.getAttribute(COUNTER);
        if (attribute == null) {
            attribute = 0;
        }
        int counter = (int) attribute;
        counter++;
        session.setAttribute(COUNTER, counter);

        // Invalidate this session if the count goes beyond 100
        if (counter >= 100) {
            session.invalidate();
        }
        return counter;
    }
}
